import torch
TRAIN_SAMPLE_PATH = r"/root/project/Code/sshcode/Transformer/news_data/train.json"
VAL_SAMPLE_PATH = r"/root/project/Code/sshcode/Transformer/news_data/val.json"

ZH_VOCAB_PATH = r"/root/project/Code/sshcode/Transformer/Transformer_model/zh_vocab.txt"
EN_VOCAB_PATH = r"/root/project/Code/sshcode/Transformer/Transformer_model/en_vocab.txt"

PAD_ID = 0 #填充的词
UNK_ID = 1
SOS_ID = 2
EOS_ID = 3

#超参数
D_MODEL = 512
N_HEAD = 8
D_FF = 2048
N = 6
DROPOUT = 0.1

BATCH_SIZE = 20
LR = 1e-5
EPOCH = 200
MAX_LEN = 50 #句子的最大长度

LABEL_SMOOTHING = 0.1 #标签平滑系数

device = "cuda" if torch.cuda.is_available() else "cpu"